from PIL import Image, ImageDraw
import os

def create_icon(size):
    # 创建一个新的图像，使用RGBA模式支持透明度
    image = Image.new('RGBA', (size, size), (0, 0, 0, 0))
    draw = ImageDraw.Draw(image)
    
    # 计算中心点和半径
    center = size // 2
    radius = int(size * 0.8) // 2
    
    # 绘制背景圆形
    draw.ellipse(
        [center - radius, center - radius, center + radius, center + radius],
        fill=(76, 175, 80, 255)  # #4CAF50
    )
    
    # 绘制大脑形状（简化的）
    brain_radius = int(radius * 0.7)
    draw.ellipse(
        [center - brain_radius, center - brain_radius, center + brain_radius, center + brain_radius],
        fill=(255, 255, 255, 255)
    )
    
    # 绘制连接线
    line_width = max(2, size // 16)
    draw.line(
        [center, center + brain_radius, center, center + radius],
        fill=(255, 255, 255, 255),
        width=line_width
    )
    
    # 绘制底部圆点
    dot_radius = max(2, size // 16)
    draw.ellipse(
        [center - dot_radius, center + radius - dot_radius,
         center + dot_radius, center + radius + dot_radius],
        fill=(255, 255, 255, 255)
    )
    
    return image

def main():
    # 创建icons目录（如果不存在）
    if not os.path.exists('icons'):
        os.makedirs('icons')
    
    # 定义需要创建的尺寸
    sizes = [16, 48, 128]
    
    # 创建不同尺寸的图标
    for size in sizes:
        icon = create_icon(size)
        icon.save(f'icons/icon{size}.png')
        print(f'Created icon{size}.png')

if __name__ == '__main__':
    main() 